//
//  UIImage+OrientationUp.m
//  CeFuBaoDemo
//
//  Created by 关旭 on 2018/11/22.
//  Copyright © 2018年 guanxu. All rights reserved.
//

#import "UIImage+OrientationUp.h"

@implementation UIImage (OrientationUp)
- (UIImage *)normalizedImage {
    if (self.imageOrientation == UIImageOrientationUp) return self;
    
    UIGraphicsBeginImageContextWithOptions(self.size, NO, self.scale);
    [self drawInRect:(CGRect){0, 0, self.size}];
    UIImage *normalizedImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    return normalizedImage;
}
- (UIImage *)fixOrientation {
    
    // No-op if the orientation is already correct
    if (self.imageOrientation == UIImageOrientationUp) return self;
    
    // We need to calculate the proper transformation to make the image upright.
    // We do it in 2 steps: Rotate if Left/Right/Down, and then flip if Mirrored.
    
    CGAffineTransform transform = CGAffineTransformIdentity;
    switch (self.imageOrientation) {
            
        case UIImageOrientationDown:
        case UIImageOrientationDownMirrored:
            transform = CGAffineTransformTranslate(transform, self.size.width, self.size.height);
            transform = CGAffineTransformRotate(transform, M_PI);
            break;
            
        case UIImageOrientationLeft:
        case UIImageOrientationLeftMirrored:
            transform = CGAffineTransformTranslate(transform, self.size.width, 0);
            transform = CGAffineTransformRotate(transform, M_PI_2);
            
            break;
            
        case UIImageOrientationRight:
        case UIImageOrientationRightMirrored:
            transform = CGAffineTransformTranslate(transform, 0, self.size.height);
            transform = CGAffineTransformRotate(transform, -M_PI_2);
            break;
            
        case UIImageOrientationUp:
        case UIImageOrientationUpMirrored:
            
            break;
            
    }
    
    switch (self.imageOrientation) {
            
        case UIImageOrientationUpMirrored:
        case UIImageOrientationDownMirrored:
            transform = CGAffineTransformTranslate(transform, self.size.width, 0);
            transform = CGAffineTransformScale(transform, -1, 1);
            break;
            
        case UIImageOrientationLeftMirrored:
        case UIImageOrientationRightMirrored:
            transform = CGAffineTransformTranslate(transform, self.size.height, 0);
            transform = CGAffineTransformScale(transform, -1, 1);
            break;
            
        case UIImageOrientationUp:
        case UIImageOrientationDown:
        case UIImageOrientationLeft:
        case UIImageOrientationRight:
            
            break;
            
    }
    
    // Now we draw the underlying CGImage into a new context, applying the transform
    // calculated above.
    
    CGContextRef ctx = CGBitmapContextCreate(NULL, self.size.width, self.size.height,
                                             CGImageGetBitsPerComponent(self.CGImage), 0,
                                             CGImageGetColorSpace(self.CGImage),
                                             CGImageGetBitmapInfo(self.CGImage));
    
    CGContextConcatCTM(ctx, transform);
    switch (self.imageOrientation) {
        case UIImageOrientationLeft:
        case UIImageOrientationLeftMirrored:
        case UIImageOrientationRight:
        case UIImageOrientationRightMirrored:
            
            // Grr...
            CGContextDrawImage(ctx, CGRectMake(0,0,self.size.height,self.size.width), self.CGImage);
            break;
            
        default:
            CGContextDrawImage(ctx, CGRectMake(0,0,self.size.width,self.size.height), self.CGImage);
            break;
    }
    
    // And now we just create a new UIImage from the drawing context
    CGImageRef cgimg = CGBitmapContextCreateImage(ctx);
    UIImage *img = [UIImage imageWithCGImage:cgimg];
    CGContextRelease(ctx);
    CGImageRelease(cgimg);
    
    return img;
    
}

//相册调用旋转90°
- (UIImage *)rotatImageWithRadius:(NSInteger)radius
{
    if (radius > 0) {
        if (radius == 90) {
            // Rigth
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:self.scale orientation:UIImageOrientationRight];
        }else if(radius == 180){
            // Down
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:self.scale orientation:UIImageOrientationDown];
        }else if(radius == 270){
            // Left
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:1.0 orientation:UIImageOrientationLeft];
        }else{
            // Up
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:1.0 orientation:UIImageOrientationUp];
        }
    }
    else
    {
        radius = ABS(radius);
        if (radius == 90) {
            // Left
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:1.0 orientation:UIImageOrientationLeft];
        }else if(radius == 180){
            // Down
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:1.0 orientation:UIImageOrientationDown];
            
        }else if(radius == 270){
            // Right
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:1.0 orientation:UIImageOrientationRight];
        }else{
            // Up
            return [[UIImage alloc] initWithCGImage:self.CGImage scale:1.0 orientation:UIImageOrientationUp];
        }
    }
}


@end
